package com.coocaa.fileexplore.utils;

import android.text.TextUtils;
import android.util.Log;

import com.coocaa.fileexplore.FileExploreHomeActivity;

import java.io.File;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Created by fc on 2017/4/12.
 */

public class FileUtils {
    public static final int SIZETYPE_B = 1;//获取文件大小单位为B的double值
    public static final int SIZETYPE_KB = 2;//获取文件大小单位为KB的double值
    public static final int SIZETYPE_MB = 3;//获取文件大小单位为MB的double值
    public static final int SIZETYPE_GB = 4;//获取文件大小单位为GB的double值
    public static final int SCAN_DEEP = 7;
    private static final Map<String, Long> cache = new HashMap<String, Long>();
    public static boolean scanFlag = false;

    /**
     * 获取文件指定文件的指定单位的大小
     *
     * @param filePath 文件路径
     * @param sizeType 获取大小的类型1为B、2为KB、3为MB、4为GB
     * @return double值的大小
     */
    public static double getFileOrFilesSize(String filePath, int sizeType) {
        if (TextUtils.isEmpty(filePath))
            return 0;
        File file = new File(filePath);
        long blockSize = 0;
        if (file.isDirectory()) {
            try {
                blockSize = getFileSizes(file, 0);
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else if (file.isFile()) {
            try {
                synchronized (cache) {
                    if (cache.containsKey(file.getAbsolutePath())) {
                        blockSize = cache.get(file.getAbsolutePath());
                    } else {
                        blockSize = getFileSize(file);
                        cache.put(file.getAbsolutePath(), blockSize);
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return FormetFileSize(blockSize, sizeType);
    }

    /**
     * 调用此方法自动计算指定文件或指定文件夹的大小
     *
     * @param filePath 文件路径
     * @return 计算好的带B、KB、MB、GB的字符串
     */
    public static String getAutoFileOrFilesSize(String filePath) {
        File file = new File(filePath);
        long blockSize = 0;
        if (file.isDirectory()) {
            try {
                blockSize = getFileSizes(file, 0);
            } catch (Exception e) {
                e.printStackTrace();
            }
        } else {
            try {
                blockSize = getFileSize(file);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return FormetFileSize(blockSize);
    }

    /**
     * 获取指定文件大小
     *
     * @param
     * @return
     * @throws Exception
     */
    private static long getFileSize(File file) throws Exception {
        long size = 0;
        if (file.exists()) {
            synchronized (cache) {
                if (cache.containsKey(file.getAbsolutePath())) {
                    size = cache.get(file.getAbsolutePath());
                } else {
                    size = file.length();
                    cache.put(file.getAbsolutePath(), size);
                }
            }
        } else {
            Log.e("获取文件大小", "文件不存在!");
            Log.e("获取文件大小", file.getPath());
        }
        return size;
    }


    /**
     * 获取指定文件夹
     *
     * @param f
     * @return
     * @throws Exception
     */
    private static long getFileSizes(File f, int deep) throws Exception {
        Log.i(FileExploreHomeActivity.TAG, "getFileSizes: " + f.getPath());
        long size = 0;
        if (deep >= SCAN_DEEP)
            return size;
        File flist[] = f.listFiles();
        if (flist == null)
            return size;
        for (int i = 0; i < flist.length; i++) {
            if (scanFlag == false)
                return 0;
            if (flist[i].isDirectory()) {
                size = size + getFileSizes(flist[i], deep + 1);
            } else if (flist[i].isFile()) {
                size = size + getFileSize(flist[i]);
            }
        }
        return size;
    }

    /**
     * 转换文件大小
     *
     * @param fileS
     * @return
     */
    private static String FormetFileSize(long fileS) {
        DecimalFormat df = new DecimalFormat("#.00");
        String fileSizeString = "";
        String wrongSize = "0B";
        if (fileS == 0) {
            return wrongSize;
        }
        if (fileS < 1024) {
            fileSizeString = df.format((double) fileS) + "B";
        } else if (fileS < 1048576) {
            fileSizeString = df.format((double) fileS / 1024) + "KB";
        } else if (fileS < 1073741824) {
            fileSizeString = df.format((double) fileS / 1048576) + "MB";
        } else {
            fileSizeString = df.format((double) fileS / 1073741824) + "GB";
        }
        return fileSizeString;
    }

    /**
     * 转换文件大小,指定转换的类型
     *
     * @param fileS
     * @param sizeType
     * @return
     */
    private static double FormetFileSize(long fileS, int sizeType) {
        DecimalFormat df = new DecimalFormat("#.000");
        double fileSizeLong = 0;
        switch (sizeType) {
            case SIZETYPE_B:
                fileSizeLong = Double.valueOf(df.format((double) fileS));
                break;
            case SIZETYPE_KB:
                fileSizeLong = Double.valueOf(df.format((double) fileS / 1024));
                break;
            case SIZETYPE_MB:
                fileSizeLong = Double.valueOf(df.format((double) fileS / 1048576));
                break;
            case SIZETYPE_GB:
                fileSizeLong = Double.valueOf(df.format((double) fileS / 1073741824));
                break;
            default:
                break;
        }
        return fileSizeLong;
    }

    public static List<File> getFileList(String path) {
        if (TextUtils.isEmpty(path))
            return null;
        File file = new File(path);
        if (file.exists()) {
            File[] files = file.listFiles();
            List<File> filterFile = new ArrayList<File>();
            if (files == null || files.length < 0)
                return null;
            Collections.addAll(filterFile, files);
            return filterFile;
        }
        return null;
    }

    public boolean isDirectory(String path) {
        if (TextUtils.isEmpty(path))
            return false;
        File file = new File(path);
        return file.isDirectory();

    }
}
